# 为了CycleFusion与其他的比较
# 重复之前D2BEGAN的过程
# 将融合模型的结果中，属于新数据集msrs-113的融合图像选取出来，作为待分析的融合图像
import os
import shutil

# 记录有多少是D（白天）
cnt = 0
# 数据集中的文件有哪些
ir_dir = r"D:\cycleFusion-project\msrs-113\src\ir"

def extract(comm_dir,target_comm_dir):
    files = os.listdir(ir_dir)      # 文件名称
    print("开始移动：")
    for dir in os.listdir(comm_dir):
        print('\n',dir,end="->")
        # 明确路径
        src_dir = os.path.join(comm_dir, dir)
        target_dir = os.path.join(target_comm_dir, dir)

        os.makedirs(os.path.join(target_comm_dir,dir),exist_ok=True)    # 创建目标路径

        for f in files:  # 将imgsNo分割成单个的文件名，split()不加空格表示用空白分割
            # 注意，这里是复制
            shutil.copyfile(os.path.join(src_dir,f), os.path.join(target_dir, f))  # 复制
            print(f.split('.')[0], end=" ")

if __name__ == '__main__':
    comm_dir = r"D:\cycleFusion-project\dataset-1444\fusedmodels\cmpPart"      # 要移动的多个Full图像集的公共父路径（源）
    target_comm_dir = r"D:\cycleFusion-project\msrs-113\fusedImgs"   # 要移动的多个Full图像集的公共父路径（目标）

    extract(comm_dir,target_comm_dir)
